import numpy as np
import matplotlib.pyplot as plt

# 创建一个示例的 numpy 矩阵
matrix = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])

# 使用 imshow 显示矩阵
plt.imshow(matrix, cmap='viridis')

# 添加颜色条
plt.colorbar()

# 显示图形
plt.show()
